/**
 * Copyright (c) 2018-2028, Chill Zhuang 庄骞 (smallchill@163.com).
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.springblade.modules.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.annotations.*;
import lombok.AllArgsConstructor;
import org.springblade.common.constant.PolicyConstant;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.secure.BladeUser;
import org.springblade.core.secure.utils.SecureUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.constant.BladeConstant;
import org.springblade.core.tool.node.INode;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.system.entity.City;
import org.springblade.modules.system.entity.Dept;
import org.springblade.modules.system.service.ICityService;
import org.springblade.modules.system.service.IDeptService;
import org.springblade.modules.system.vo.AVueTableVo;
import org.springblade.modules.system.vo.DeptVO;
import org.springblade.modules.system.wrapper.CityWrapper;
import org.springblade.modules.system.wrapper.DeptWrapper;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 控制器
 *
 * @author Chill
 */
@RestController
@AllArgsConstructor
@RequestMapping("/blade-system/dept")
@ApiIgnore
@Api(value = "部门", tags = "部门")
public class DeptController extends BladeController {

	private IDeptService deptService;

	private ICityService cityService;

	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "详情", notes = "传入dept")
	public R<DeptVO> detail(Dept dept) {
		Dept detail = deptService.getOne(Condition.getQueryWrapper(dept));
		return R.data(DeptWrapper.build().entityVO(detail));
	}

	/**
	 * 列表
	 */
	@GetMapping("/list")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "deptName", value = "部门名称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "fullName", value = "部门全称", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "列表", notes = "传入dept")
	public R<List<INode>> list(@ApiIgnore @RequestParam Map<String, Object> dept, BladeUser bladeUser) {
		QueryWrapper<Dept> queryWrapper = Condition.getQueryWrapper(dept, Dept.class);
		List<Dept> list = deptService.list((!bladeUser.getTenantId().equals(BladeConstant.ADMIN_TENANT_ID)) ? queryWrapper.lambda().eq(Dept::getTenantId, bladeUser.getTenantId()) : queryWrapper);
		return R.data(DeptWrapper.build().listNodeVO(list));
	}

	/**
	 * 懒加载列表
	 */
	@GetMapping("/lazy-list")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "short", value = "部门简称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "name", value = "部门名称", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "懒加载列表", notes = "传入dept")
	public R<List<INode>> lazyList(Long cityId, Long parentId, @ApiIgnore @RequestParam Map<String, Object> param) {
		List<INode> list = deptService.lazyList(cityId, parentId, param);
		return R.data(CityWrapper.build().listNodeLazyVO(list));
	}

	/**
	 * 懒加载列表
	 */
	@GetMapping("/lazy-tree")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "short", value = "部门简称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "name", value = "部门名称", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "懒加载列表", notes = "传入dept")
	public R<List<INode>> lazyTree(Long parentId, @ApiIgnore @RequestParam Map<String, Object> param) {
		List<INode> list = deptService.lazyTree(parentId, param);
		return R.data(DeptWrapper.build().listNodeLazyVO(list));
	}

	/**
	 * 获取部门树形结构
	 *
	 * @return
	 */
	@GetMapping("/tree")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "树形结构", notes = "树形结构")
	public R<List<INode>> recursionTree(Long cityId) {
		List<INode> tree = deptService.recursionTree(cityId);
		return R.data(tree);
	}

	/**
	 * 新增或修改
	 */
	@PostMapping("/submit")
	@ApiOperationSupport(order = 6)
	@ApiOperation(value = "新增或修改", notes = "传入dept")
	public R submit(@Valid @RequestBody Dept dept) {
		return R.status(deptService.saveOrUpdate(dept));
	}

	/**
	 * 删除
	 */
	@PostMapping("/remove")
	@ApiOperationSupport(order = 7)
	@ApiOperation(value = "删除", notes = "传入ids")
	public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
		return R.status(deptService.removeByIds(Func.toLongList(ids)));
	}

	/*
	 根据发文层级获取相应发文机关
 	*/
	@GetMapping("/level/list")
	public R getDeptByLevel(@RequestParam Integer level) {
		if (level == -1){
			return null;
		}
		LambdaQueryWrapper<City> cityLambdaQueryWrapper = Wrappers.lambdaQuery();
		if (level == PolicyConstant.POLOCY_LEVEL_COUNTY){
			cityLambdaQueryWrapper.ge(City::getRegionLevel,level);
		}else {
			cityLambdaQueryWrapper.eq(City::getRegionLevel,level);
		}
		List<City> list = cityService.list(cityLambdaQueryWrapper);
		List<Long> cityIds = list.stream().map(City::getId).collect(Collectors.toList());
		if (cityIds.size() == 0){
			return R.fail("所选层级下无发文机关，请重新选择");
		}
		LambdaQueryWrapper<Dept> deptLambdaQueryWrapper = Wrappers.lambdaQuery();
		deptLambdaQueryWrapper.in(Dept::getCityId,cityIds);
		List<Dept> list1 = deptService.list(deptLambdaQueryWrapper);
		List<AVueTableVo> tables = new ArrayList<>(16);
		list1.forEach(i->{
			AVueTableVo aVueTableVo = new AVueTableVo();
			aVueTableVo.setLabel(i.getDeptName());
			aVueTableVo.setValue(i.getId()+"");
			tables.add(aVueTableVo);
		});
		if (list1.size() == 0) {
			return R.fail("所选层级下无发文机关，请重新选择");
		} else {
			return R.data(tables);
		}
	}

}
